0692c4a4722c70035376acd578f0779d3a0ae523,hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/TestDistributedRT.java,TestDistributedRT,testNMHeartbeat,#,238
Before Change
MockNM nm1, nm2;
//Wait for leader election to start
Thread.sleep(3000);
if (rm1.getRMContext().getGroupMembershipService().isLeader()) {
rmL = rm1;
rmRT = rm2;
} else {
After Change
MockNM nm1, nm2;
//Wait for leader election to start
Thread.sleep(3000);
if (rm1.getRMContext().isLeader()) {
rmL = rm1;
rmRT = rm2;
} else {